if(ITK_WRAP_PYTHON)
  set(test_input_dir ${itk-module_SOURCE_DIR}/test/Input)

  itk_python_add_test(
      NAME
      PythonLazyLoadingImage
      COMMAND
      LazyLoadingImageTest.py
  )

  list(FIND ITK_WRAP_IMAGE_DIMS 2 wrap_2_index)
  if(ITK_WRAP_float AND wrap_2_index GREATER -1)
    # TODO: reenable the geodesic test once we get why the result is not the
    # same than with c++
    # keep just one to be sure it run, but don't compare the images

    itk_python_add_test(
          NAME
          PythonGeodesicActiveContourLeftVentricleTest
          #     --compare GeodesicActiveContourLeftVentricleTest.png
          #             DATA{${WrapITK_SOURCE_DIR}/images/GeodesicActiveContourLeftVentricleTest.png}
          COMMAND
          GeodesicActiveContourImageFilterTest.py
          DATA{${test_input_dir}/BrainProtonDensitySlice.png}
          ${ITK_TEST_OUTPUT_DIR}/GeodesicActiveContourLeftVentricleTest.png
          81
          114
          5.0
          1.0
          -0.5
          3.0
          2.0
    )

    itk_python_add_test(
          NAME
          PythonThresholdSegmentationLevelSetVentricleTest
          TEST_DRIVER_ARGS
          --compare
          ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetVentricleTest.png
          DATA{Baseline/PythonThresholdSegmentationLevelSetVentricleTest.png}
          COMMAND
          ThresholdSegmentationLevelSetImageFilterTest.py
          DATA{${test_input_dir}/BrainProtonDensitySlice.png}
          ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetVentricleTest.png
          81
          112
          5
          210
          250
    )

    itk_python_add_test(
          NAME
          PythonThresholdSegmentationLevelSetWhiteMatterTest
          TEST_DRIVER_ARGS
          --compare
          ${ITK_TEST_OUTPUT_DIR}/ThresholdSegmentationLevelSetWhiteMatterTest.png
          DATA{Baseline/PythonThresholdSegmentationLevelSetWhiteMatterTest.png}
          COMMAND
          ThresholdSegmentationLevelSetImageFilterTest.py
          DATA{${test_input_dir}/BrainProtonDensitySlice.png}
          ${ITK_TEST_OUTPUT_DIR}/ThresholdSegmentationLevelSetWhiteMatterTest.png
          60
          116
          5
          150
          180
    )

    itk_python_add_test(
          NAME
          PythonThresholdSegmentationLevelSetGrayMatterTest
          TEST_DRIVER_ARGS
          --compare
          ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetGrayMatterTest.png
          DATA{Baseline/PythonThresholdSegmentationLevelSetGrayMatterTest.png}
          COMMAND
          ThresholdSegmentationLevelSetImageFilterTest.py
          DATA{${WrapITK_SOURCE_DIR}/images/BrainProtonDensitySlice.png}
          ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetGrayMatterTest.png
          107
          69
          5
          180
          210
    )

    itk_python_expression_add_test(
          NAME
          itkBinaryMaskToNarrowBandPointSetFilterPythonTest
          EXPRESSION
          "filt = itk.BinaryMaskToNarrowBandPointSetFilter.New()"
    )
  endif()
endif()
